package modals

import (
	"github.com/joho/godotenv"
	"log"
	"os"
)

// InEnv InEnv.
type InEnv struct {
	InfluxDBAddr     string
	InfluxDBToken    string
	InfluxDBDatabase string
	InfluxDBOrg      string
}

// TgEnv TgEnv.
type TgEnv struct {
	TgToken    string
	TgChanelID string
}

// LoadConfigInflux функция получения данных из файла .env
func LoadConfigInflux() InEnv {
	err := godotenv.Load(".env")
	if err != nil {
		log.Println(err)
	}
	e := InEnv{
		InfluxDBAddr:     os.Getenv("INFLUX_DB_ADDR"),
		InfluxDBToken:    os.Getenv("INFLUX_DB_TOKEN"),
		InfluxDBDatabase: os.Getenv("INFLUX_DB_BAC"),
		InfluxDBOrg:      os.Getenv("INFLUX_DB_ORG"),
	}
	return e
}

// LoadConfigTelegram функция получения данных из файла.env
func LoadConfigTelegram() TgEnv {
	err := godotenv.Load(".env")
	if err != nil {
		log.Println(err)
	}
	e := TgEnv{
		TgToken:    os.Getenv("TG_TOKEN"),
		TgChanelID: os.Getenv("TG_CHANEL_ID"),
	}
	return e
}
